import os

# 检查双色球目录中的.pth文件
ssq_dir = "scripts/ssq"
pth_files = [f for f in os.listdir(ssq_dir) if f.endswith('.pth')]

print("双色球目录中的.pth文件:")
for file in pth_files:
    print(f"  {file}")
    # 检查文件大小
    file_path = os.path.join(ssq_dir, file)
    size = os.path.getsize(file_path)
    print(f"    大小: {size} 字节")

# 检查增强版LSTM-CRF模型文件是否存在
enhanced_model_path = os.path.join(ssq_dir, "enhanced_lstm_crf_model.pth")
if os.path.exists(enhanced_model_path):
    print(f"\n增强版LSTM-CRF模型文件存在: {enhanced_model_path}")
    size = os.path.getsize(enhanced_model_path)
    print(f"  文件大小: {size} 字节")
else:
    print(f"\n增强版LSTM-CRF模型文件不存在: {enhanced_model_path}")

# 检查基础LSTM模型文件是否存在
lstm_model_path = os.path.join(ssq_dir, "ssq_lstm_model.pth")
if os.path.exists(lstm_model_path):
    print(f"基础LSTM模型文件存在: {lstm_model_path}")
    size = os.path.getsize(lstm_model_path)
    print(f"  文件大小: {size} 字节")
else:
    print(f"基础LSTM模型文件不存在: {lstm_model_path}")